python program for compound interest